#coding=utf-8
'''
# @Name:output
# @author:mysouil
# @Description:
# @DATE:2022/9/21 10:08
# @Modified By:
'''
from json import dumps

from flask import current_app, make_response
from flask_restful.utils import PY3


def output_json(data, code, headers=None):
    """Makes a Flask response with a JSON encoded body"""
    
    # 此处添加自己定义的json格式规则
    if 'message' not in data:
        d = {
            'code': 200,
            'message': 'OK',
            'data': data
        }
    else:
        d = data
    
    settings = current_app.config.get('RESTFUL_JSON', {})
    
    # If we're in debug mode, and the indent is not set, we set it to a
    # reasonable value here.  Note that this won't override any existing value
    # that was set.  We also set the "sort_keys" value.
    if current_app.debug:
        settings.setdefault('indent', 4)
        settings.setdefault('sort_keys', not PY3)
    
    # always end the json dumps with a new line
    # see https://github.com/mitsuhiko/flask/pull/1262
    dumped = dumps(d, **settings) + "\n"
    
    resp = make_response(dumped, code)
    resp.headers.extend(headers or {})
    return resp